package tests;

import common.CheapCompany;
import common.Company;
import common.ExpensiveCompany;
import common.MediumCompany;
import common.enumCompany;

import junit.framework.TestCase;

public class CompanyTest extends TestCase {

	Company expensive;
	Company medium;
	Company cheap;
	
	protected void setUp() {
		expensive = new ExpensiveCompany(enumCompany.PHOENIX);
		medium = new MediumCompany(enumCompany.HYDRA);
		cheap = new CheapCompany(enumCompany.ZETA);
		expensive.increaseSize(4);
		cheap.increaseSize(14);
		medium.increaseSize(25);
	}

	protected void tearDown() {
		
	}
	
	public void testAddToBoard() { 
		expensive.addToBoard();
		assertTrue(expensive.isOnBoard());
		assertTrue(!cheap.isOnBoard());
	}
	
	public void testGetFirstPrizeValue() {
		assertEquals(6000,expensive.getFirstPrizeValue());
		assertEquals(9000,medium.getFirstPrizeValue());
		assertEquals(7000,cheap.getFirstPrizeValue());
	}
	
	public void testToString() { 
		assertEquals("Hydra", medium.toString());
	}
	
	public void testIncreaseSize() { 
		medium.increaseSize(10);
		assertEquals(35,medium.getSize());
	}
	
	
 }
